import os
def mydirsize(d1):
    dlist=os.listdir(d1)
    dirsize=0
    for f in dlist:
        f1=os.path.join(d1,f)
        if os.path.isfile(f1):
            dirsize=dirsize+os.path.getsize(f1)
        if os.path.isdir(f1):
            dirsize=dirsize+mydirsize(f1)
    return dirsize




dirstr=input('请输入目录名：')
print('-'*50)
if os.path.isdir(dirstr):
	print('{}目录大小为（B）：{}'.format(dirstr,mydirsize(dirstr)))
else:
	print('%s不是目录名！统计失败！'%dirstr)